import cv2
from matplotlib import pyplot as plt
import pandas as pd
fig = plt.figure(figsize=(10, 7))
rows = 2
columns = 2
df = pd.read_csv("dataset.csv")
df
print(len(df['Label']))
switch = {
1: "SuknelÄ—",
2: "Å varkas",
3: "Trumpas topas",
4: "Ilgas topas",
5: "Sijonas",
6: "Suknele be rankovių",
7: "SkarelÄ—",
8: "Paltas",
9: "Striuke",
10: "Marškiniai",
11: "KelnÄ—s"
}
# max 122
skip = 0
load = 121
columns = 5
fig = plt.figure(figsize=(28, load))
rows = int(load / 4)
for i in range(load):
index = i + skip
fig.add_subplot(rows, columns, index + 1)
plt.imshow(cv2.imread("imgs/" + df['Path'][index]))
plt.axis('off')
plt.title(switch.get(int(df['Label'][index])))